package problem.year2021.february;

/**
 * 给定 n 个整数，找出平均数最大且长度为 k 的连续子数组，并输出该最大平均数。
 * <p>
 * 示例：
 * <p>
 * 输入：[1,12,-5,-6,50,3], k = 4
 * 输出：12.75
 * 解释：最大平均数 (12-5-6+50)/4 = 51/4 = 12.75
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= k <= n <= 30,000。
 * 所给数据范围 [-10,000，10,000]。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/maximum-average-subarray-i
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class February4 {
    // 简单的优点过分，滑动窗口，平均去头
    public double findMaxAverage(int[] nums, int k) {
        int sum = 0;
        for (int i = 0; i < k; ++i) {
            sum += nums[i];
        }
        int sum2 = sum;
        // 1<=k<=n
        for (int i = k; i < nums.length; ++i) {
            sum2 = sum2 - nums[i - k] + nums[i];
            sum = Math.max(sum, sum2);
        }
        return sum * 1.0 / k;
    }

    public static void main(String[] args) {
        // * 输入：[1,12,-5,-6,50,3], k = 4
        // * 输出：12.75
        int[] nums = new int[]{1, 12, -5, -6, 50, 3};
        System.out.println(new February4().findMaxAverage(nums, 4));
        double k = (12 - 5 - 6 + 50) / (4 * 1.0);
        System.out.println(k);
    }
}
